# 统计指定目录大小
# 引入os模块
import os

# 定义函数
def count_dirsize(dir_obj):
    '''
    统计指定目录大小函数
    :param dir_obj
    :return: 目录文件大小总和
    '''
    # 初始化文件大小和
    Sum = 0
    # 获取指定目录中的所有文件信息
    dlist = os.listdir(dir_obj)
    # 遍历所有文件并判断
    for f in dlist:
        # 为遍历的文件添加目录路径
        file = os.path.join(dir_obj,f)
        # 判断是否是文件，是文件读取大小并求和
        if os.path.isfile(file):
            Sum += os.path.getsize(file)  # 调用求文件大小函数
        # 判断是否是目录（文件夹）
        if os.path.isdir(file):
            Sum += count_dirsize(file)  # 递归调用自己，来统计子目录的大小

    # 注意返回目录大小值
    return Sum

# 输入要统计的目录
dir_obj = input('不加引号输入目录路径: ')
# 输出大小
print('该目录大小为:',count_dirsize(dir_obj),'字节')